import pymysql
import time

while True:
    # 直接连入students库
    db = pymysql.connect(host='49.233.39.160', user='user', database='lebo16', port=3306,
                         password='leboAa!#$123', charset='utf8')

    # 获取游标
    cur = db.cursor()

    czxh = int(input('你好,登录系统，1：注册系统  2：登录系统  3：退出  请输入对应的操作序号：\n'))


    # 注册
    if czxh == 1:
        print('**************开始注册**************')
        user1 = str(input('请输入姓名：\n'))
        password1 = int(input('请输入学号：\n'))
        # 判断信息是否已被注册
        sql = "select * from students_zbddhz where studentno = %s;" % password1
        aa = cur.execute(sql)
        db.commit()
        if aa == 1:
            print("用户名已存在，请重新注册")
        else:
            sql = "insert into students_zbddhz(name,studentNo) values (%s, %d);" % (user1, password1)
            cur.execute(sql)

            db.commit()
            print("注册成功")
        db.close()

    # 登陆
    elif czxh == 2:
        print('**************开始登录**************')
        zh = str(input("请输入姓名：\n"))
        mm = int(input("请输入学号：\n"))

        # 数据库表中查询是否含有该姓名和编号
        sql = "select * from students_zbddhz where  (name = %s) and (studentno = %d)" % (zh, mm)
        aa = cur.execute(sql)
        db.commit()

        if aa == 1:
            print("登录成功")

        else:
            print("账号或密码错误")
        db.close()

    elif czxh == 3:
        print("退出系统")
        break
    # 其他
    else:
        print("输入的命令不正确")
        db.close()
